package lc20240513;


import java.util.HashSet;
import java.util.Set;

//54. 有效的数独

public class Demo1 {
    public static void main(String[] args) {
        char[][] board = {
                {'5', '3', '.', '.', '7', '.', '.', '.', '.'},
                {'6', '.', '.', '1', '9', '5', '.', '.', '.'},
                {'.', '9', '8', '.', '.', '.', '.', '6', '.'},
                {'8', '.', '.', '.', '6', '.', '.', '.', '3'},
                {'4', '.', '.', '8', '.', '3', '.', '.', '1'},
                {'7', '.', '.', '.', '2', '.', '.', '.', '6'},
                {'.', '6', '.', '.', '.', '.', '2', '8', '.'},
                {'.', '.', '.', '4', '1', '9', '.', '.', '5'},
                {'.', '.', '.', '.', '8', '.', '.', '7', '9'}
        };
        System.out.println("Is valid sudoku: " + isValidSudoku1(board));
    }

    public static boolean isValidSudoku(char[][] board) {
        Set<Character>[] rows = new HashSet[9];
        Set<Character>[] cols = new HashSet[9];
        Set<Character>[][] boxes = new HashSet[3][3];

        // 初始化 HashSet
        for (int i = 0; i < 9; i++) {
            //建立三个set，横着的，竖着的，还有九个小九宫格。
            rows[i] = new HashSet<>();
            cols[i] = new HashSet<>();
            boxes[i / 3][i % 3] = new HashSet<>();
        }

        // 遍历数独板
        for (int r = 0; r < 9; r++) {
            for (int c = 0; c < 9; c++) {
                char num = board[r][c];
                if (num == '.') {
                    continue; // 跳过空格
                }
                if (rows[r].contains(num)) {
                    return false; // 检查行
                }
                if (cols[c].contains(num)) {
                    return false; // 检查列
                }
                if (boxes[r / 3][c / 3].contains(num)) {
                    return false; // 检查3x3小格
                }
                // 将该数添加到对应的 HashSet 中
                rows[r].add(num);
                cols[c].add(num);
                boxes[r / 3][c / 3].add(num);
            }
        }
        return true;
    }

    public static boolean isValidSudoku1(char[][] board) {
        int[][] cols = new int[9][9];
        int[][] rows = new int[9][9];
        int[][][] boxes = new int[3][3][9];

        for (int i = 0; i < 9; i++) {
            for (int j = 0; j < 9; j++) {
                char c = board[i][j];
                if (c != '.') {
                    int index = c - '0' - 1;
                    cols[j][index]++;
                    rows[i][index]++;
                    boxes[i / 3][j / 3][index]++;
                    if (cols[j][index] > 1 || rows[i][index] > 1 || boxes[i / 3][j / 3][index] > 1) {
                        return false;
                    }
                }
            }
        }
        return true;
    }
}
